from kyb_cap import driver
# 截图方法小括号中可以设置截图到指定目录，设值一个img_name变量,值为路径加文件名，就可以做到
driver.get_screenshot_as_file(r'login1.png')
driver.save_screenshot(r'login2.png')

driver.quit()